cs201p assignment 1 correct solution